#!/usr/bin/env python
# coding=utf-8
'''
FilePath     : /websocket-demo/client.py
Description  :  
Author       : desyang
Date         : 2025-02-24 10:40:53
LastEditors  : desyang
LastEditTime : 2025-02-24 15:37:37
'''
from socketio import Client
from utils import generate_qrcode, DOMAIN

# 初始化客户端
sio = Client()

# 定义事件处理函数
@sio.event
def connect():
    print("已连接到服务器")


generate_qrcode(f'http://{DOMAIN}/download/test_file_name')

@sio.event
def download_triggered(data):
    sio.disconnect()
    print(f"下载已触发！文件ID: {data['file_id']}")
    # 在这里添加处理逻辑

# 连接到服务器
file_id = "test_file_name"  # 替换为二维码中的文件

sio.connect(f"http://{DOMAIN}?file_id={file_id}")

# 保持连接
try:
    # 保持连接
    sio.wait()
except KeyboardInterrupt:
    print("捕获到 Ctrl+C，正在断开连接...")
    sio.disconnect()
    print("已断开连接")